from vilipixiv.ygy.ygy_sd_api import *
import csv

with open('../csv/tag_count_1.csv', 'r', encoding='UTF-8') as f:
    reader = csv.reader(f)
    column_data = []
    for row in reader:
        column_data.append(row[1])

set_model(0)
text2img_save(str(column_data), 3, 768, 512, '../ai_img/')
